package com.ljy.my_study.java.zip;

import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * @author lijunying
 * @date 2021/11/2 17:50
 */
public class ZipCompress {

    /**
     * 目的地Zip文件
     */
    private String zipFileName;

    //创建zip输出流
    private ZipOutputStream out;

    //创建缓冲输出流
    private BufferedOutputStream bos;

    public ZipCompress(String zipFileName) throws FileNotFoundException {
        this.zipFileName=zipFileName;

        //创建zip输出流
        out = new ZipOutputStream( new FileOutputStream(zipFileName));

        //创建缓冲输出流
        bos = new BufferedOutputStream(out);
    }

    public void zip(File... files) throws Exception {
        for(File f:files){
            zip(f);
        }
    }

    public void zip(File sourceFile) throws Exception {
        //调用函数
        compress(out,bos,sourceFile,sourceFile.getName());
    }

    public void compress(ZipOutputStream out,BufferedOutputStream bos,File sourceFile,String base) throws Exception
    {
        //如果路径为目录（文件夹）
        if(sourceFile.isDirectory())
        {

            //取出文件夹中的文件（或子文件夹）
            File[] flist = sourceFile.listFiles();

            if(flist.length==0)//如果文件夹为空，则只需在目的地zip文件中写入一个目录进入点
            {
                System.out.println(base+"/");
                out.putNextEntry(  new ZipEntry(base+"/") );
            }
            else//如果文件夹不为空，则递归调用compress，文件夹中的每一个文件（或文件夹）进行压缩
            {
                for(int i=0;i<flist.length;i++)
                {
                    compress(out,bos,flist[i],base+"/"+flist[i].getName());
                }
            }
        }
        else//如果不是目录（文件夹），即为文件，则先写入目录进入点，之后将文件写入zip文件中
        {
            out.putNextEntry( new ZipEntry(base) );
            FileInputStream fos = new FileInputStream(sourceFile);
            BufferedInputStream bis = new BufferedInputStream(fos);

            int tag;
            System.out.println(base);
            //将源文件写入到zip文件中
            while((tag=bis.read())!=-1){
                bos.write(tag);
            }
            bis.close();
            fos.close();

        }
    }

    public void close() throws IOException {
        bos.close();
        out.close();
        System.out.println("压缩完成");
    }
}
